home *** CD-ROM | disk | FTP | other *** search
Wrap
# Source Generated with Decompyle++ # File: in.pyc (Python 2.4) from test.test_support import verify, vereq, verbose, TestFailed, TESTFN, get_original_stdout from copy import deepcopy import warnings warnings.filterwarnings('ignore', 'complex divmod\\(\\), // and % are deprecated$', DeprecationWarning, '(<string>|%s)$' % __name__) def veris(a, b): if a is not b: raise TestFailed, '%r is %r' % (a, b) def testunop(a, res, expr = 'len(a)', meth = '__len__'): if verbose: print 'checking', expr dict = { 'a': a } vereq(eval(expr, dict), res) t = type(a) m = getattr(t, meth) while meth not in t.__dict__: t = t.__bases__[0] vereq(m, t.__dict__[meth]) vereq(m(a), res) bm = getattr(a, meth) vereq(bm(), res) def testbinop(a, b, res, expr = 'a+b', meth = '__add__'): if verbose: print 'checking', expr dict = { 'a': a, 'b': b } if meth == '__div__' and 1 / 2 == 0.5: meth = '__truediv__' vereq(eval(expr, dict), res) t = type(a) m = getattr(t, meth) while meth not in t.__dict__: t = t.__bases__[0] vereq(m, t.__dict__[meth]) vereq(m(a, b), res) bm = getattr(a, meth) vereq(bm(b), res) def testternop(a, b, c, res, expr = 'a[b:c]', meth = '__getslice__'): if verbose: print 'checking', expr dict = { 'a': a, 'b': b, 'c': c } vereq(eval(expr, dict), res) t = type(a) m = getattr(t, meth) while meth not in t.__dict__: t = t.__bases__[0] vereq(m, t.__dict__[meth]) vereq(m(a, b, c), res) bm = getattr(a, meth) vereq(bm(b, c), res) def testsetop(a, b, res, stmt = 'a+=b', meth = '__iadd__'): if verbose: print 'checking', stmt dict = { 'a': deepcopy(a), 'b': b } exec stmt in dict vereq(dict['a'], res) t = type(a) m = getattr(t, meth) while meth not in t.__dict__: t = t.__bases__[0] vereq(m, t.__dict__[meth]) dict['a'] = deepcopy(a) m(dict['a'], b) vereq(dict['a'], res) dict['a'] = deepcopy(a) bm = getattr(dict['a'], meth) bm(b) vereq(dict['a'], res) def testset2op(a, b, c, res, stmt = 'a[b]=c', meth = '__setitem__'): if verbose: print 'checking', stmt dict = { 'a': deepcopy(a), 'b': b, 'c': c } exec stmt in dict vereq(dict['a'], res) t = type(a) m = getattr(t, meth) while meth not in t.__dict__: t = t.__bases__[0] vereq(m, t.__dict__[meth]) dict['a'] = deepcopy(a) m(dict['a'], b, c) vereq(dict['a'], res) dict['a'] = deepcopy(a) bm = getattr(dict['a'], meth) bm(b, c) vereq(dict['a'], res) def testset3op(a, b, c, d, res, stmt = 'a[b:c]=d', meth = '__setslice__'): if verbose: print 'checking', stmt dict = { 'a': deepcopy(a), 'b': b, 'c': c, 'd': d } exec stmt in dict vereq(dict['a'], res) t = type(a) while meth not in t.__dict__: t = t.__bases__[0] m = getattr(t, meth) vereq(m, t.__dict__[meth]) dict['a'] = deepcopy(a) m(dict['a'], b, c, d) vereq(dict['a'], res) dict['a'] = deepcopy(a) bm = getattr(dict['a'], meth) bm(b, c, d) vereq(dict['a'], res) def class_docstrings(): class Classic: '''A classic docstring.''' pass vereq(Classic.__doc__, 'A classic docstring.') vereq(Classic.__dict__['__doc__'], 'A classic docstring.') class Classic2: pass verify(Classic2.__doc__ is None) class NewStatic(object): '''Another docstring.''' pass vereq(NewStatic.__doc__, 'Another docstring.') vereq(NewStatic.__dict__['__doc__'], 'Another docstring.') class NewStatic2(object): pass verify(NewStatic2.__doc__ is None) class NewDynamic(object): '''Another docstring.''' pass vereq(NewDynamic.__doc__, 'Another docstring.') vereq(NewDynamic.__dict__['__doc__'], 'Another docstring.') class NewDynamic2(object): pass verify(NewDynamic2.__doc__ is None) def lists(): if verbose: print 'Testing list operations...' testbinop([ 1], [ 2], [ 1, 2], 'a+b', '__add__') testbinop([ 1, 2, 3], 2, 1, 'b in a', '__contains__') testbinop([ 1, 2, 3], 4, 0, 'b in a', '__contains__') testbinop([ 1, 2, 3], 1, 2, 'a[b]', '__getitem__') testternop([ 1, 2, 3], 0, 2, [ 1, 2], 'a[b:c]', '__getslice__') testsetop([ 1], [ 2], [ 1, 2], 'a+=b', '__iadd__') testsetop([ 1, 2], 3, [ 1, 2, 1, 2, 1, 2], 'a*=b', '__imul__') testunop([ 1, 2, 3], 3, 'len(a)', '__len__') testbinop([ 1, 2], 3, [ 1, 2, 1, 2, 1, 2], 'a*b', '__mul__') testbinop([ 1, 2], 3, [ 1, 2, 1, 2, 1, 2], 'b*a', '__rmul__') testset2op([ 1, 2], 1, 3, [ 1, 3], 'a[b]=c', '__setitem__') testset3op([ 1, 2, 3, 4], 1, 3, [ 5, 6], [ 1, 5, 6, 4], 'a[b:c]=d', '__setslice__') def dicts(): if verbose: print 'Testing dict operations...' testbinop({ 1: 2 }, { 2: 1 }, -1, 'cmp(a,b)', '__cmp__') testbinop({ 1: 2, 3: 4 }, 1, 1, 'b in a', '__contains__') testbinop({ 1: 2, 3: 4 }, 2, 0, 'b in a', '__contains__') testbinop({ 1: 2, 3: 4 }, 1, 2, 'a[b]', '__getitem__') d = { 1: 2, 3: 4 } l1 = [] for i in d.keys(): l1.append(i) l = [] for i in iter(d): l.append(i) vereq(l, l1) l = [] for i in d.__iter__(): l.append(i) vereq(l, l1) l = [] for i in dict.__iter__(d): l.append(i) vereq(l, l1) d = { 1: 2, 3: 4 } testunop(d, 2, 'len(a)', '__len__') vereq(eval(repr(d), { }), d) vereq(eval(d.__repr__(), { }), d) testset2op({ 1: 2, 3: 4 }, 2, 3, { 1: 2, 2: 3, 3: 4 }, 'a[b]=c', '__setitem__') def dict_constructor(): if verbose: print 'Testing dict constructor ...' d = dict() vereq(d, { }) d = dict({ }) vereq(d, { }) d = dict({ 1: 2, 'a': 'b' }) vereq(d, { 1: 2, 'a': 'b' }) vereq(d, dict(d.items())) vereq(d, dict(d.iteritems())) d = dict({ 'one': 1, 'two': 2 }) vereq(d, dict(one = 1, two = 2)) vereq(d, dict(**d)) vereq(d, dict({ 'one': 1 }, two = 2)) vereq(d, dict([ ('two', 2)], one = 1)) vereq(d, dict([ ('one', 100), ('two', 200)], **d)) verify(d is not dict(**d)) for badarg in (0, 0x0L, (0.0+0.0j), '0', [ 0], (0,)): try: dict(badarg) except TypeError: continue except ValueError: if badarg == '0': pass else: raise TestFailed('no TypeError from dict(%r)' % badarg) badarg == '0' else: raise TestFailed('no TypeError from dict(%r)' % badarg) try: dict({ }, { }) except TypeError: pass raise TestFailed('no TypeError from dict({}, {})') class Mapping: dict = { 1: 2, 3: 4, 'a': (0.0+1.0j) } try: dict(Mapping()) except TypeError: pass raise TestFailed('no TypeError from dict(incomplete mapping)') Mapping.keys = lambda self: self.dict.keys() Mapping.__getitem__ = lambda self, i: self.dict[i] d = dict(Mapping()) vereq(d, Mapping.dict) class AddressBookEntry: def __init__(self, first, last): self.first = first self.last = last def __iter__(self): return iter([ self.first, self.last]) d = dict([ AddressBookEntry('Tim', 'Warsaw'), AddressBookEntry('Barry', 'Peters'), AddressBookEntry('Tim', 'Peters'), AddressBookEntry('Barry', 'Warsaw')]) vereq(d, { 'Barry': 'Warsaw', 'Tim': 'Peters' }) d = dict(zip(range(4), range(1, 5))) dict([], []([ (i, i + 1) for i in range(4) ])) for bad in ([ ('tooshort',)], [ ('too', 'long', 'by 1')]): try: dict(bad) except ValueError: d d vereq continue raise TestFailed('no ValueError from dict(%r)' % bad) def test_dir(): if verbose: print 'Testing dir() ...' junk = 12 vereq(dir(), [ 'junk']) del junk for arg in (2, 0x2L, (0.0+2.0j), 2.0, [ 2], '2', u'2', (2,), { 2: 2 }, type, test_dir): dir(arg) class C: Cdata = 1 def Cmethod(self): pass cstuff = [ 'Cdata', 'Cmethod', '__doc__', '__module__'] vereq(dir(C), cstuff) verify('im_self' in dir(C.Cmethod)) c = C() vereq(dir(c), cstuff) c.cdata = 2 c.cmethod = lambda self: 0 vereq(dir(c), cstuff + [ 'cdata', 'cmethod']) verify('im_self' in dir(c.Cmethod)) class A(C): Adata = 1 def Amethod(self): pass astuff = [ 'Adata', 'Amethod'] + cstuff vereq(dir(A), astuff) verify('im_self' in dir(A.Amethod)) a = A() vereq(dir(a), astuff) verify('im_self' in dir(a.Amethod)) a.adata = 42 a.amethod = lambda self: 3 vereq(dir(a), astuff + [ 'adata', 'amethod']) def interesting(strings): return _[1] class C(object): Cdata = 1 def Cmethod(self): pass cstuff = [ 'Cdata', 'Cmethod'] vereq(interesting(dir(C)), cstuff) c = C() vereq(interesting(dir(c)), cstuff) verify('im_self' in dir(C.Cmethod)) c.cdata = 2 c.cmethod = lambda self: 0 vereq(interesting(dir(c)), cstuff + [ 'cdata', 'cmethod']) verify('im_self' in dir(c.Cmethod)) class A(C): Adata = 1 def Amethod(self): pass astuff = [ 'Adata', 'Amethod'] + cstuff vereq(interesting(dir(A)), astuff) verify('im_self' in dir(A.Amethod)) a = A() vereq(interesting(dir(a)), astuff) a.adata = 42 a.amethod = lambda self: 3 vereq(interesting(dir(a)), astuff + [ 'adata', 'amethod']) verify('im_self' in dir(a.Amethod)) import sys class M(type(sys)): pass minstance = M('m') minstance.b = 2 minstance.a = 1 names = _[1] vereq(names, [ 'a', 'b']) class M2(M): def getdict(self): return 'Not a dict!' __dict__ = property(getdict) m2instance = M2('m2') m2instance.b = 2 m2instance.a = 1 vereq(m2instance.__dict__, 'Not a dict!') try: dir(m2instance) except TypeError: [] [] [] except: [] vereq(dir(None), dir(Ellipsis)) class Wrapper(object): def __init__(self, obj): self._Wrapper__obj = obj def __repr__(self): return 'Wrapper(%s)' % repr(self._Wrapper__obj) def __getitem__(self, key): return Wrapper(self._Wrapper__obj[key]) def __len__(self): return len(self._Wrapper__obj) def __getattr__(self, name): return Wrapper(getattr(self._Wrapper__obj, name)) class C(object): def __getclass(self): return Wrapper(type(self)) __class__ = property(__getclass) dir(C()) binops = { 'add': '+', 'sub': '-', 'mul': '*', 'div': '/', 'mod': '%', 'divmod': 'divmod', 'pow': '**', 'lshift': '<<', 'rshift': '>>', 'and': '&', 'xor': '^', 'or': '|', 'cmp': 'cmp', 'lt': '<', 'le': '<=', 'eq': '==', 'ne': '!=', 'gt': '>', 'ge': '>=' } for name, expr in binops.items(): if expr.islower(): expr = expr + '(a, b)' else: expr = 'a %s b' % expr binops[name] = expr unops = { 'pos': '+', 'neg': '-', 'abs': 'abs', 'invert': '~', 'int': 'int', 'long': 'long', 'float': 'float', 'oct': 'oct', 'hex': 'hex' } for name, expr in unops.items(): if expr.islower(): expr = expr + '(a)' else: expr = '%s a' % expr unops[name] = expr def numops(a, b, skip = []): dict = { 'a': a, 'b': b } for name, expr in binops.items(): if name not in skip: name = '__%s__' % name if hasattr(a, name): res = eval(expr, dict) testbinop(a, b, res, expr, name) hasattr(a, name) for name, expr in unops.items(): if name not in skip: name = '__%s__' % name if hasattr(a, name): res = eval(expr, dict) testunop(a, res, expr, name) hasattr(a, name) def ints(): if verbose: print 'Testing int operations...' numops(100, 3) vereq(1.__nonzero__(), 1) vereq(0.__nonzero__(), 0) class C(int): def __add__(self, other): return NotImplemented vereq(C(0x5L), 5) try: C() + '' except TypeError: pass raise TestFailed, 'NotImplemented should have caused TypeError' import sys try: C(sys.maxint + 1) except OverflowError: pass raise TestFailed, 'should have raised OverflowError' def longs(): if verbose: print 'Testing long operations...' numops(0x64L, 0x3L) def floats(): if verbose: print 'Testing float operations...' numops(100.0, 3.0) def complexes(): if verbose: print 'Testing complex operations...' numops((0.0+100.0j), (0.0+3.0j), skip = [ 'lt', 'le', 'gt', 'ge', 'int', 'long', 'float']) class Number(complex): __slots__ = [ 'prec'] def __new__(cls, *args, **kwds): result = complex.__new__(cls, *args) result.prec = kwds.get('prec', 12) return result def __repr__(self): prec = self.prec if self.imag == 0.0: return '%.*g' % (prec, self.real) if self.real == 0.0: return '%.*gj' % (prec, self.imag) return '(%.*g+%.*gj)' % (prec, self.real, prec, self.imag) __str__ = __repr__ a = Number(3.1400000000000001, prec = 6) vereq(repr(a), '3.14') vereq(a.prec, 6) a = Number(a, prec = 2) vereq(repr(a), '3.1') vereq(a.prec, 2) a = Number(234.5) vereq(repr(a), '234.5') vereq(a.prec, 12) def spamlists(): if verbose: print 'Testing spamlist operations...' import copy import xxsubtype as spam def spamlist(l, memo = None): import xxsubtype as spam return spam.spamlist(l) copy._deepcopy_dispatch[spam.spamlist] = spamlist testbinop(spamlist([ 1]), spamlist([ 2]), spamlist([ 1, 2]), 'a+b', '__add__') testbinop(spamlist([ 1, 2, 3]), 2, 1, 'b in a', '__contains__') testbinop(spamlist([ 1, 2, 3]), 4, 0, 'b in a', '__contains__') testbinop(spamlist([ 1, 2, 3]), 1, 2, 'a[b]', '__getitem__') testternop(spamlist([ 1, 2, 3]), 0, 2, spamlist([ 1, 2]), 'a[b:c]', '__getslice__') testsetop(spamlist([ 1]), spamlist([ 2]), spamlist([ 1, 2]), 'a+=b', '__iadd__') testsetop(spamlist([ 1, 2]), 3, spamlist([ 1, 2, 1, 2, 1, 2]), 'a*=b', '__imul__') testunop(spamlist([ 1, 2, 3]), 3, 'len(a)', '__len__') testbinop(spamlist([ 1, 2]), 3, spamlist([ 1, 2, 1, 2, 1, 2]), 'a*b', '__mul__') testbinop(spamlist([ 1, 2]), 3, spamlist([ 1, 2, 1, 2, 1, 2]), 'b*a', '__rmul__') testset2op(spamlist([ 1, 2]), 1, 3, spamlist([ 1, 3]), 'a[b]=c', '__setitem__') testset3op(spamlist([ 1, 2, 3, 4]), 1, 3, spamlist([ 5, 6]), spamlist([ 1, 5, 6, 4]), 'a[b:c]=d', '__setslice__') class C(spam.spamlist): def foo(self): return 1 a = C() vereq(a, []) vereq(a.foo(), 1) a.append(100) vereq(a, [ 100]) vereq(a.getstate(), 0) a.setstate(42) vereq(a.getstate(), 42) def spamdicts(): if verbose: print 'Testing spamdict operations...' import copy import xxsubtype as spam def spamdict(d, memo = None): import xxsubtype as spam sd = spam.spamdict() for k, v in d.items(): sd[k] = v return sd copy._deepcopy_dispatch[spam.spamdict] = spamdict testbinop(spamdict({ 1: 2 }), spamdict({ 2: 1 }), -1, 'cmp(a,b)', '__cmp__') testbinop(spamdict({ 1: 2, 3: 4 }), 1, 1, 'b in a', '__contains__') testbinop(spamdict({ 1: 2, 3: 4 }), 2, 0, 'b in a', '__contains__') testbinop(spamdict({ 1: 2, 3: 4 }), 1, 2, 'a[b]', '__getitem__') d = spamdict({ 1: 2, 3: 4 }) l1 = [] for i in d.keys(): l1.append(i) l = [] for i in iter(d): l.append(i) vereq(l, l1) l = [] for i in d.__iter__(): l.append(i) vereq(l, l1) l = [] for i in type(spamdict({ })).__iter__(d): l.append(i) vereq(l, l1) straightd = { 1: 2, 3: 4 } spamd = spamdict(straightd) testunop(spamd, 2, 'len(a)', '__len__') testunop(spamd, repr(straightd), 'repr(a)', '__repr__') testset2op(spamdict({ 1: 2, 3: 4 }), 2, 3, spamdict({ 1: 2, 2: 3, 3: 4 }), 'a[b]=c', '__setitem__') class C(spam.spamdict): def foo(self): return 1 a = C() vereq(a.items(), []) vereq(a.foo(), 1) a['foo'] = 'bar' vereq(a.items(), [ ('foo', 'bar')]) vereq(a.getstate(), 0) a.setstate(100) vereq(a.getstate(), 100) def pydicts(): if verbose: print 'Testing Python subclass of dict...' verify(issubclass(dict, dict)) verify(isinstance({ }, dict)) d = dict() vereq(d, { }) verify(d.__class__ is dict) verify(isinstance(d, dict)) class C(dict): state = -1 def __init__(self, *a, **kw): if a: vereq(len(a), 1) self.state = a[0] if kw: for k, v in kw.items(): self[v] = k def __getitem__(self, key): return self.get(key, 0) def __setitem__(self, key, value): verify(isinstance(key, type(0))) dict.__setitem__(self, key, value) def setstate(self, state): self.state = state def getstate(self): return self.state verify(issubclass(C, dict)) a1 = C(12) vereq(a1.state, 12) a2 = C(foo = 1, bar = 2) if a2[1] == 'foo': pass vereq(a2[2], 'bar') a = C() vereq(a.state, -1) vereq(a.getstate(), -1) a.setstate(0) vereq(a.state, 0) vereq(a.getstate(), 0) a.setstate(10) vereq(a.state, 10) vereq(a.getstate(), 10) vereq(a[42], 0) a[42] = 24 vereq(a[42], 24) if verbose: print 'pydict stress test ...' N = 50 for i in range(N): a[i] = C() for j in range(N): a[i][j] = i * j for i in range(N): for j in range(N): vereq(a[i][j], i * j) def pylists(): if verbose: print 'Testing Python subclass of list...' class C(list): def __getitem__(self, i): return list.__getitem__(self, i) + 100 def __getslice__(self, i, j): return (i, j) a = C() a.extend([ 0, 1, 2]) vereq(a[0], 100) vereq(a[1], 101) vereq(a[2], 102) vereq(a[100:200], (100, 200)) def metaclass(): if verbose: print 'Testing __metaclass__...' class C: __metaclass__ = type def __init__(self): self._C__state = 0 def getstate(self): return self._C__state def setstate(self, state): self._C__state = state a = C() vereq(a.getstate(), 0) a.setstate(10) vereq(a.getstate(), 10) class D: class __metaclass__(type): def myself(cls): return cls vereq(D.myself(), D) d = D() verify(d.__class__ is D) class M1(type): def __new__(cls, name, bases, dict): dict['__spam__'] = 1 return type.__new__(cls, name, bases, dict) class C: __metaclass__ = M1 vereq(C.__spam__, 1) c = C() vereq(c.__spam__, 1) class _instance(object): pass class M2(object): def __new__(cls, name, bases, dict): self = object.__new__(cls) self.name = name self.bases = bases self.dict = dict return self __new__ = staticmethod(__new__) def __call__(self): it = _instance() for key in self.dict: if key.startswith('__'): continue setattr(it, key, self.dict[key].__get__(it, self)) return it class C: __metaclass__ = M2 def spam(self): return 42 vereq(C.name, 'C') vereq(C.bases, ()) verify('spam' in C.dict) c = C() vereq(c.spam(), 42) class autosuper(type): def __new__(metaclass, name, bases, dict): cls = super(autosuper, metaclass).__new__(metaclass, name, bases, dict) while name[:1] == '_': name = name[1:] if name: name = '_%s__super' % name else: name = '__super' setattr(cls, name, super(cls)) return cls class A: __metaclass__ = autosuper def meth(self): return 'A' class B(A): def meth(self): return 'B' + self._B__super.meth() class C(A): def meth(self): return 'C' + self._C__super.meth() class D(C, B): def meth(self): return 'D' + self._D__super.meth() vereq(D().meth(), 'DCBA') class E(B, C): def meth(self): return 'E' + self._E__super.meth() vereq(E().meth(), 'EBCA') class autoproperty(type): def __new__(metaclass, name, bases, dict): hits = { } for key, val in dict.iteritems(): if key.startswith('_get_'): key = key[5:] (get, set) = hits.get(key, (None, None)) get = val hits[key] = (get, set) continue if key.startswith('_set_'): key = key[5:] (get, set) = hits.get(key, (None, None)) set = val hits[key] = (get, set) continue for get, set in hits.iteritems(): dict[key] = property(get, set) return super(autoproperty, metaclass).__new__(metaclass, name, bases, dict) class A: __metaclass__ = autoproperty def _get_x(self): return -(self._A__x) def _set_x(self, x): self._A__x = -x a = A() verify(not hasattr(a, 'x')) a.x = 12 vereq(a.x, 12) vereq(a._A__x, -12) class multimetaclass(autoproperty, autosuper): pass class A: __metaclass__ = multimetaclass def _get_x(self): return 'A' class B(A): def _get_x(self): return 'B' + self._B__super._get_x() class C(A): def _get_x(self): return 'C' + self._C__super._get_x() class D(C, B): def _get_x(self): return 'D' + self._D__super._get_x() vereq(D().x, 'DCBA') class T(type): counter = 0 def __init__(self, *args): T.counter += 1 class C: __metaclass__ = T vereq(T.counter, 1) a = C() vereq(type(a), C) vereq(T.counter, 1) class C(object): pass c = C() try: c() except TypeError: pass raise TestFailed, 'calling object w/o call method should raise TypeError' def pymods(): if verbose: print 'Testing Python subclass of module...' log = [] import sys MT = type(sys) class MM(MT): def __init__(self, name): MT.__init__(self, name) def __getattribute__(self, name): log.append(('getattr', name)) return MT.__getattribute__(self, name) def __setattr__(self, name, value): log.append(('setattr', name, value)) MT.__setattr__(self, name, value) def __delattr__(self, name): log.append(('delattr', name)) MT.__delattr__(self, name) a = MM('a') a.foo = 12 x = a.foo del a.foo vereq(log, [ ('setattr', 'foo', 12), ('getattr', 'foo'), ('delattr', 'foo')]) def multi(): if verbose: print 'Testing multiple inheritance...' class C(object): def __init__(self): self._C__state = 0 def getstate(self): return self._C__state def setstate(self, state): self._C__state = state a = C() vereq(a.getstate(), 0) a.setstate(10) vereq(a.getstate(), 10) class D(dict, C): def __init__(self): type({ }).__init__(self) C.__init__(self) d = D() vereq(d.keys(), []) d['hello'] = 'world' vereq(d.items(), [ ('hello', 'world')]) vereq(d['hello'], 'world') vereq(d.getstate(), 0) d.setstate(10) vereq(d.getstate(), 10) vereq(D.__mro__, (D, dict, C, object)) class Node(object): def __int__(self): return int(self.foo()) def foo(self): return '23' class Frag(Node, list): def foo(self): return '42' vereq(Node().__int__(), 23) vereq(int(Node()), 23) vereq(Frag().__int__(), 42) vereq(int(Frag()), 42) class A: x = 1 class B(A): pass class C(A): x = 2 class D(B, C): pass vereq(D.x, 1) class E(D, object): pass vereq(E.__mro__, (E, D, B, A, C, object)) vereq(E.x, 1) class F(B, C, object): pass vereq(F.__mro__, (F, B, C, A, object)) vereq(F.x, 2) class C: def cmethod(self): return 'C a' def all_method(self): return 'C b' class M1(C, object): def m1method(self): return 'M1 a' def all_method(self): return 'M1 b' vereq(M1.__mro__, (M1, C, object)) m = M1() vereq(m.cmethod(), 'C a') vereq(m.m1method(), 'M1 a') vereq(m.all_method(), 'M1 b') class D(C): def dmethod(self): return 'D a' def all_method(self): return 'D b' class M2(D, object): def m2method(self): return 'M2 a' def all_method(self): return 'M2 b' vereq(M2.__mro__, (M2, D, C, object)) m = M2() vereq(m.cmethod(), 'C a') vereq(m.dmethod(), 'D a') vereq(m.m2method(), 'M2 a') vereq(m.all_method(), 'M2 b') class M3(M1, M2, object): def m3method(self): return 'M3 a' def all_method(self): return 'M3 b' vereq(M3.__mro__, (M3, M1, M2, D, C, object)) m = M3() vereq(m.cmethod(), 'C a') vereq(m.dmethod(), 'D a') vereq(m.m1method(), 'M1 a') vereq(m.m2method(), 'M2 a') vereq(m.m3method(), 'M3 a') vereq(m.all_method(), 'M3 b') class Classic: pass try: class New(Classic): __metaclass__ = type except TypeError: pass raise TestFailed, "new class with only classic bases - shouldn't be" def diamond(): if verbose: print 'Testing multiple inheritance special cases...' class A(object): def spam(self): return 'A' vereq(A().spam(), 'A') class B(A): def boo(self): return 'B' def spam(self): return 'B' vereq(B().spam(), 'B') vereq(B().boo(), 'B') class C(A): def boo(self): return 'C' vereq(C().spam(), 'A') vereq(C().boo(), 'C') class D(B, C): pass vereq(D().spam(), 'B') vereq(D().boo(), 'B') vereq(D.__mro__, (D, B, C, A, object)) class E(C, B): pass vereq(E().spam(), 'B') vereq(E().boo(), 'C') vereq(E.__mro__, (E, C, B, A, object)) try: class F(D, E): pass except TypeError: pass raise TestFailed, 'expected MRO order disagreement (F)' try: class G(E, D): pass except TypeError: pass raise TestFailed, 'expected MRO order disagreement (G)' def ex5(): if verbose: print 'Testing ex5 from C3 switch discussion...' class A(object): pass class B(object): pass class C(object): pass class X(A): pass class Y(A): pass class Z(X, B, Y, C): pass vereq(Z.__mro__, (Z, X, B, Y, A, C, object)) def monotonicity(): if verbose: print 'Testing MRO monotonicity...' class Boat(object): pass class DayBoat(Boat): pass class WheelBoat(Boat): pass class EngineLess(DayBoat): pass class SmallMultihull(DayBoat): pass class PedalWheelBoat(EngineLess, WheelBoat): pass class SmallCatamaran(SmallMultihull): pass class Pedalo(PedalWheelBoat, SmallCatamaran): pass vereq(PedalWheelBoat.__mro__, (PedalWheelBoat, EngineLess, DayBoat, WheelBoat, Boat, object)) vereq(SmallCatamaran.__mro__, (SmallCatamaran, SmallMultihull, DayBoat, Boat, object)) vereq(Pedalo.__mro__, (Pedalo, PedalWheelBoat, EngineLess, SmallCatamaran, SmallMultihull, DayBoat, WheelBoat, Boat, object)) def consistency_with_epg(): if verbose: print 'Testing consistentcy with EPG...' class Pane(object): pass class ScrollingMixin(object): pass class EditingMixin(object): pass class ScrollablePane(Pane, ScrollingMixin): pass class EditablePane(Pane, EditingMixin): pass class EditableScrollablePane(ScrollablePane, EditablePane): pass vereq(EditableScrollablePane.__mro__, (EditableScrollablePane, ScrollablePane, EditablePane, Pane, ScrollingMixin, EditingMixin, object)) mro_err_msg = 'Cannot create a consistent method resolution\norder (MRO) for bases ' def mro_disagreement(): if verbose: print 'Testing error messages for MRO disagreement...' def raises(exc, expected, callable, *args): try: callable(*args) except exc: msg = None if not str(msg).startswith(expected): raise TestFailed, 'Message %r, expected %r' % (str(msg), expected) except: str(msg).startswith(expected) raise TestFailed, 'Expected %s' % exc class A(object): pass class B(A): pass class C(object): pass raises(TypeError, 'duplicate base class A', type, 'X', (A, A), { }) raises(TypeError, mro_err_msg, type, 'X', (A, B), { }) raises(TypeError, mro_err_msg, type, 'X', (A, C, B), { }) class GridLayout(object): pass class HorizontalGrid(GridLayout): pass class VerticalGrid(GridLayout): pass class HVGrid(HorizontalGrid, VerticalGrid): pass class VHGrid(VerticalGrid, HorizontalGrid): pass raises(TypeError, mro_err_msg, type, 'ConfusedGrid', (HVGrid, VHGrid), { }) def objects(): if verbose: print 'Testing object class...' a = object() vereq(a.__class__, object) vereq(type(a), object) b = object() verify(a is not b) verify(not hasattr(a, 'foo')) try: a.foo = 12 except (AttributeError, TypeError): pass verify(0, 'object() should not allow setting a foo attribute') verify(not hasattr(object(), '__dict__')) class Cdict(object): pass x = Cdict() vereq(x.__dict__, { }) x.foo = 1 vereq(x.foo, 1) vereq(x.__dict__, { 'foo': 1 }) def slots(): if verbose: print 'Testing __slots__...' class C0(object): __slots__ = [] x = C0() verify(not hasattr(x, '__dict__')) verify(not hasattr(x, 'foo')) class C1(object): __slots__ = [ 'a'] x = C1() verify(not hasattr(x, '__dict__')) verify(not hasattr(x, 'a')) x.a = 1 vereq(x.a, 1) x.a = None veris(x.a, None) del x.a verify(not hasattr(x, 'a')) class C3(object): __slots__ = [ 'a', 'b', 'c'] x = C3() verify(not hasattr(x, '__dict__')) verify(not hasattr(x, 'a')) verify(not hasattr(x, 'b')) verify(not hasattr(x, 'c')) x.a = 1 x.b = 2 x.c = 3 vereq(x.a, 1) vereq(x.b, 2) vereq(x.c, 3) class C4(object): '''Validate name mangling''' __slots__ = [ '__a'] def __init__(self, value): self._C4__a = value def get(self): return self._C4__a x = C4(5) verify(not hasattr(x, '__dict__')) verify(not hasattr(x, '__a')) vereq(x.get(), 5) try: x.__a = 6 except AttributeError: pass raise TestFailed, 'Double underscored names not mangled' try: class C(object): __slots__ = [ None] except TypeError: pass raise TestFailed, '[None] slots not caught' try: class C(object): __slots__ = [ 'foo bar'] except TypeError: pass raise TestFailed, "['foo bar'] slots not caught" try: class C(object): __slots__ = [ 'foo\x00bar'] except TypeError: pass raise TestFailed, "['foo\\0bar'] slots not caught" try: class C(object): __slots__ = [ '1'] except TypeError: pass raise TestFailed, "['1'] slots not caught" try: class C(object): __slots__ = [ ''] except TypeError: pass raise TestFailed, "[''] slots not caught" class C(object): __slots__ = [ 'a', 'a_b', '_a', 'A0123456789Z'] class Counted(object): counter = 0 def __init__(self): Counted.counter += 1 def __del__(self): Counted.counter -= 1 class C(object): __slots__ = [ 'a', 'b', 'c'] x = C() x.a = Counted() x.b = Counted() x.c = Counted() vereq(Counted.counter, 3) del x vereq(Counted.counter, 0) class D(C): pass x = D() x.a = Counted() x.z = Counted() vereq(Counted.counter, 2) del x vereq(Counted.counter, 0) class E(D): __slots__ = [ 'e'] x = E() x.a = Counted() x.z = Counted() x.e = Counted() vereq(Counted.counter, 3) del x vereq(Counted.counter, 0) class F(object): __slots__ = [ 'a', 'b'] log = [] s = F() s.a = [ Counted(), s] vereq(Counted.counter, 1) s = None import gc gc.collect() vereq(Counted.counter, 0) import sys import gc class G(object): def __cmp__(self, other): return 0 g = G() orig_objects = len(gc.get_objects()) for i in xrange(10): g == g new_objects = len(gc.get_objects()) vereq(orig_objects, new_objects) class H(object): __slots__ = [ 'a', 'b'] def __init__(self): self.a = 1 self.b = 2 def __del__(self): if not self.a == 1: raise AssertionError if not self.b == 2: raise AssertionError save_stderr = sys.stderr sys.stderr = sys.stdout h = H() try: del h finally: sys.stderr = save_stderr def slotspecials(): if verbose: print 'Testing __dict__ and __weakref__ in __slots__...' class D(object): __slots__ = [ '__dict__'] a = D() verify(hasattr(a, '__dict__')) verify(not hasattr(a, '__weakref__')) a.foo = 42 vereq(a.__dict__, { 'foo': 42 }) class W(object): __slots__ = [ '__weakref__'] a = W() verify(hasattr(a, '__weakref__')) verify(not hasattr(a, '__dict__')) try: a.foo = 42 except AttributeError: pass raise TestFailed, "shouldn't be allowed to set a.foo" class C1(W, D): __slots__ = [] a = C1() verify(hasattr(a, '__dict__')) verify(hasattr(a, '__weakref__')) a.foo = 42 vereq(a.__dict__, { 'foo': 42 }) class C2(D, W): __slots__ = [] a = C2() verify(hasattr(a, '__dict__')) verify(hasattr(a, '__weakref__')) a.foo = 42 vereq(a.__dict__, { 'foo': 42 }) def dynamics(): if verbose: print 'Testing class attribute propagation...' class D(object): pass class E(D): pass class F(D): pass D.foo = 1 vereq(D.foo, 1) vereq(E.foo, 1) vereq(F.foo, 1) class C(object): pass a = C() verify(not hasattr(a, 'foobar')) C.foobar = 2 vereq(a.foobar, 2) C.method = lambda self: 42 vereq(a.method(), 42) C.__repr__ = lambda self: 'C()' vereq(repr(a), 'C()') C.__int__ = lambda self: 100 vereq(int(a), 100) vereq(a.foobar, 2) verify(not hasattr(a, 'spam')) def mygetattr(self, name): if name == 'spam': return 'spam' raise AttributeError C.__getattr__ = mygetattr vereq(a.spam, 'spam') a.new = 12 vereq(a.new, 12) def mysetattr(self, name, value): if name == 'spam': raise AttributeError return object.__setattr__(self, name, value) C.__setattr__ = mysetattr try: a.spam = 'not spam' except AttributeError: pass verify(0, 'expected AttributeError') vereq(a.spam, 'spam') class D(C): pass d = D() d.foo = 1 vereq(d.foo, 1) class I(int): pass vereq('a' * I(2), 'aa') vereq(I(2) * 'a', 'aa') vereq(2 * I(3), 6) vereq(I(3) * 2, 6) vereq(I(3) * I(2), 6) class L(long): pass vereq('a' * L(0x2L), 'aa') vereq(L(0x2L) * 'a', 'aa') vereq(2 * L(3), 6) vereq(L(3) * 2, 6) vereq(L(3) * L(2), 6) class dynamicmetaclass(type): pass class someclass: __metaclass__ = dynamicmetaclass verify(someclass != object) def errors(): if verbose: print 'Testing errors...' try: class C(list, dict): pass except TypeError: pass verify(0, 'inheritance from both list and dict should be illegal') try: class C(object, None): pass except TypeError: pass verify(0, 'inheritance from non-type should be illegal') class Classic: pass try: class C(type(len)): pass except TypeError: pass verify(0, 'inheritance from CFunction should be illegal') try: class C(object): __slots__ = 1 except TypeError: pass verify(0, '__slots__ = 1 should be illegal') try: class C(object): __slots__ = [ 1] except TypeError: pass verify(0, '__slots__ = [1] should be illegal') def classmethods(): if verbose: print 'Testing class methods...' class C(object): def foo(*a): return a goo = classmethod(foo) c = C() vereq(C.goo(1), (C, 1)) vereq(c.goo(1), (C, 1)) vereq(c.foo(1), (c, 1)) class D(C): pass d = D() vereq(D.goo(1), (D, 1)) vereq(d.goo(1), (D, 1)) vereq(d.foo(1), (d, 1)) vereq(D.foo(d, 1), (d, 1)) def f(cls, arg): return (cls, arg) ff = classmethod(f) vereq(ff.__get__(0, int)(42), (int, 42)) vereq(ff.__get__(0)(42), (int, 42)) veris(C.goo.im_self, C) veris(D.goo.im_self, D) veris(super(D, D).goo.im_self, D) veris(super(D, d).goo.im_self, D) vereq(super(D, D).goo(), (D,)) vereq(super(D, d).goo(), (D,)) try: classmethod(1).__get__(1) except TypeError: pass raise TestFailed, 'classmethod should check for callability' def classmethods_in_c(): if verbose: print 'Testing C-based class methods...' import xxsubtype as spam a = (1, 2, 3) d = { 'abc': 123 } (x, a1, d1) = spam.spamlist.classmeth(*a, **d) veris(x, spam.spamlist) vereq(a, a1) vereq(d, d1) (x, a1, d1) = spam.spamlist().classmeth(*a, **d) veris(x, spam.spamlist) vereq(a, a1) vereq(d, d1) def staticmethods(): if verbose: print 'Testing static methods...' class C(object): def foo(*a): return a goo = staticmethod(foo) c = C() vereq(C.goo(1), (1,)) vereq(c.goo(1), (1,)) vereq(c.foo(1), (c, 1)) class D(C): pass d = D() vereq(D.goo(1), (1,)) vereq(d.goo(1), (1,)) vereq(d.foo(1), (d, 1)) vereq(D.foo(d, 1), (d, 1)) def staticmethods_in_c(): if verbose: print 'Testing C-based static methods...' import xxsubtype as spam a = (1, 2, 3) d = { 'abc': 123 } (x, a1, d1) = spam.spamlist.staticmeth(*a, **d) veris(x, None) vereq(a, a1) vereq(d, d1) (x, a1, d2) = spam.spamlist().staticmeth(*a, **d) veris(x, None) vereq(a, a1) vereq(d, d1) def classic(): if verbose: print 'Testing classic classes...' class C: def foo(*a): return a goo = classmethod(foo) c = C() vereq(C.goo(1), (C, 1)) vereq(c.goo(1), (C, 1)) vereq(c.foo(1), (c, 1)) class D(C): pass d = D() vereq(D.goo(1), (D, 1)) vereq(d.goo(1), (D, 1)) vereq(d.foo(1), (d, 1)) vereq(D.foo(d, 1), (d, 1)) class E: foo = C.foo vereq(E().foo, C.foo) verify(repr(C.foo.__get__(C())).startswith('<bound method ')) def compattr(): if verbose: print 'Testing computed attributes...' class C(object): class computed_attribute(object): def __init__(self, get, set = None, delete = None): self._computed_attribute__get = get self._computed_attribute__set = set self._computed_attribute__delete = delete def __get__(self, obj, type = None): return self._computed_attribute__get(obj) def __set__(self, obj, value): return self._computed_attribute__set(obj, value) def __delete__(self, obj): return self._computed_attribute__delete(obj) def __init__(self): self._C__x = 0 def __get_x(self): x = self._C__x self._C__x = x + 1 return x def __set_x(self, x): self._C__x = x def __delete_x(self): del self._C__x x = computed_attribute(__get_x, __set_x, __delete_x) a = C() vereq(a.x, 0) vereq(a.x, 1) a.x = 10 vereq(a.x, 10) vereq(a.x, 11) del a.x vereq(hasattr(a, 'x'), 0) def newslot(): if verbose: print 'Testing __new__ slot override...' class C(list): def __new__(cls): self = list.__new__(cls) self.foo = 1 return self def __init__(self): self.foo = self.foo + 2 a = C() vereq(a.foo, 3) verify(a.__class__ is C) class D(C): pass b = D() vereq(b.foo, 3) verify(b.__class__ is D) def altmro(): if verbose: print 'Testing mro() and overriding it...' class A(object): def f(self): return 'A' class B(A): pass class C(A): def f(self): return 'C' class D(B, C): pass vereq(D.mro(), [ D, B, C, A, object]) vereq(D.__mro__, (D, B, C, A, object)) vereq(D().f(), 'C') class PerverseMetaType(type): def mro(cls): L = type.mro(cls) L.reverse() return L class X(D, B, C, A): __metaclass__ = PerverseMetaType vereq(X.__mro__, (object, A, C, B, D, X)) vereq(X().f(), 'A') def overloading(): if verbose: print 'Testing operator overloading...' class B(object): """Intermediate class because object doesn't have a __setattr__""" pass class C(B): def __getattr__(self, name): if name == 'foo': return ('getattr', name) else: raise AttributeError def __setattr__(self, name, value): if name == 'foo': self.setattr = (name, value) else: return B.__setattr__(self, name, value) def __delattr__(self, name): if name == 'foo': self.delattr = name else: return B.__delattr__(self, name) def __getitem__(self, key): return ('getitem', key) def __setitem__(self, key, value): self.setitem = (key, value) def __delitem__(self, key): self.delitem = key def __getslice__(self, i, j): return ('getslice', i, j) def __setslice__(self, i, j, value): self.setslice = (i, j, value) def __delslice__(self, i, j): self.delslice = (i, j) a = C() vereq(a.foo, ('getattr', 'foo')) a.foo = 12 vereq(a.setattr, ('foo', 12)) del a.foo vereq(a.delattr, 'foo') vereq(a[12], ('getitem', 12)) a[12] = 21 vereq(a.setitem, (12, 21)) del a[12] vereq(a.delitem, 12) vereq(a[0:10], ('getslice', 0, 10)) a[0:10] = 'foo' vereq(a.setslice, (0, 10, 'foo')) del a[0:10] vereq(a.delslice, (0, 10)) def methods(): if verbose: print 'Testing methods...' class C(object): def __init__(self, x): self.x = x def foo(self): return self.x c1 = C(1) vereq(c1.foo(), 1) class D(C): boo = C.foo goo = c1.foo d2 = D(2) vereq(d2.foo(), 2) vereq(d2.boo(), 2) vereq(d2.goo(), 1) class E(object): foo = C.foo vereq(E().foo, C.foo) verify(repr(C.foo.__get__(C(1))).startswith('<bound method ')) def specials(): if verbose: print 'Testing special operators...' class C(object): def __getitem__(self, i): if i <= i: pass elif i < 10: return i raise IndexError c1 = C() c2 = C() verify(not (not c1)) vereq(hash(c1), id(c1)) vereq(cmp(c1, c2), cmp(id(c1), id(c2))) vereq(c1, c1) verify(c1 != c2) verify(not (c1 != c1)) verify(not (c1 == c2)) verify(str(c1).find('C object at ') >= 0) vereq(str(c1), repr(c1)) verify(-1 not in c1) for i in range(10): verify(i in c1) verify(10 not in c1) class D(object): def __getitem__(self, i): if i <= i: pass elif i < 10: return i raise IndexError d1 = D() d2 = D() verify(not (not d1)) vereq(hash(d1), id(d1)) vereq(cmp(d1, d2), cmp(id(d1), id(d2))) vereq(d1, d1) verify(d1 != d2) verify(not (d1 != d1)) verify(not (d1 == d2)) verify(str(d1).find('D object at ') >= 0) vereq(str(d1), repr(d1)) verify(-1 not in d1) for i in range(10): verify(i in d1) verify(10 not in d1) class Proxy(object): def __init__(self, x): self.x = x def __nonzero__(self): return not (not (self.x)) def __hash__(self): return hash(self.x) def __eq__(self, other): return self.x == other def __ne__(self, other): return self.x != other def __cmp__(self, other): return cmp(self.x, other.x) def __str__(self): return 'Proxy:%s' % self.x def __repr__(self): return 'Proxy(%r)' % self.x def __contains__(self, value): return value in self.x p0 = Proxy(0) p1 = Proxy(1) p_1 = Proxy(-1) verify(not p0) verify(not (not p1)) vereq(hash(p0), hash(0)) vereq(p0, p0) verify(p0 != p1) verify(not (p0 != p0)) vereq(not p0, p1) vereq(cmp(p0, p1), -1) vereq(cmp(p0, p0), 0) vereq(cmp(p0, p_1), 1) vereq(str(p0), 'Proxy:0') vereq(repr(p0), 'Proxy(0)') p10 = Proxy(range(10)) verify(-1 not in p10) for i in range(10): verify(i in p10) verify(10 not in p10) class DProxy(object): def __init__(self, x): self.x = x def __nonzero__(self): return not (not (self.x)) def __hash__(self): return hash(self.x) def __eq__(self, other): return self.x == other def __ne__(self, other): return self.x != other def __cmp__(self, other): return cmp(self.x, other.x) def __str__(self): return 'DProxy:%s' % self.x def __repr__(self): return 'DProxy(%r)' % self.x def __contains__(self, value): return value in self.x p0 = DProxy(0) p1 = DProxy(1) p_1 = DProxy(-1) verify(not p0) verify(not (not p1)) vereq(hash(p0), hash(0)) vereq(p0, p0) verify(p0 != p1) verify(not (p0 != p0)) vereq(not p0, p1) vereq(cmp(p0, p1), -1) vereq(cmp(p0, p0), 0) vereq(cmp(p0, p_1), 1) vereq(str(p0), 'DProxy:0') vereq(repr(p0), 'DProxy(0)') p10 = DProxy(range(10)) verify(-1 not in p10) for i in range(10): verify(i in p10) verify(10 not in p10) def unsafecmp(a, b): try: a.__class__.__cmp__(a, b) except TypeError: pass raise TestFailed, "shouldn't allow %s.__cmp__(%r, %r)" % (a.__class__, a, b) unsafecmp(u'123', '123') unsafecmp('123', u'123') unsafecmp(1, 1.0) unsafecmp(1.0, 1) unsafecmp(1, 0x1L) unsafecmp(0x1L, 1) class Letter(str): def __new__(cls, letter): if letter == 'EPS': return str.__new__(cls) return str.__new__(cls, letter) def __str__(self): if not self: return 'EPS' return self import sys test_stdout = sys.stdout sys.stdout = get_original_stdout() try: print Letter('w') except RuntimeError: pass raise TestFailed, 'expected a RuntimeError for print recursion' sys.stdout = test_stdout def weakrefs(): if verbose: print 'Testing weak references...' import weakref class C(object): pass c = C() r = weakref.ref(c) verify(r() is c) del c verify(r() is None) del r class NoWeak(object): __slots__ = [ 'foo'] no = NoWeak() try: weakref.ref(no) except TypeError: msg = None verify(str(msg).find('weak reference') >= 0) verify(0, 'weakref.ref(no) should be illegal') class Weak(object): __slots__ = [ 'foo', '__weakref__'] yes = Weak() r = weakref.ref(yes) verify(r() is yes) del yes verify(r() is None) del r def properties(): if verbose: print 'Testing property...' class C(object): def getx(self): return self._C__x def setx(self, value): self._C__x = value def delx(self): del self._C__x x = property(getx, setx, delx, doc = "I'm the x property.") a = C() verify(not hasattr(a, 'x')) a.x = 42 vereq(a._C__x, 42) vereq(a.x, 42) del a.x verify(not hasattr(a, 'x')) verify(not hasattr(a, '_C__x')) C.x.__set__(a, 100) vereq(C.x.__get__(a), 100) C.x.__delete__(a) verify(not hasattr(a, 'x')) raw = C.__dict__['x'] verify(isinstance(raw, property)) attrs = dir(raw) verify('__doc__' in attrs) verify('fget' in attrs) verify('fset' in attrs) verify('fdel' in attrs) vereq(raw.__doc__, "I'm the x property.") verify(raw.fget is C.__dict__['getx']) verify(raw.fset is C.__dict__['setx']) verify(raw.fdel is C.__dict__['delx']) for attr in ('__doc__', 'fget', 'fset', 'fdel'): try: setattr(raw, attr, 42) except TypeError: msg = None if str(msg).find('readonly') < 0: raise TestFailed('when setting readonly attr %r on a property, got unexpected TypeError msg %r' % (attr, str(msg))) str(msg).find('readonly') < 0 raise TestFailed('expected TypeError from trying to set readonly %r attr on a property' % attr) class D(object): __getitem__ = property((lambda s: 1 / 0)) d = D() try: for i in d: str(i) except ZeroDivisionError: pass raise TestFailed, 'expected ZeroDivisionError from bad property' def supers(): if verbose: print 'Testing super...' class A(object): def meth(self, a): return 'A(%r)' % a vereq(A().meth(1), 'A(1)') class B(A): def __init__(self): self._B__super = super(B, self) def meth(self, a): return 'B(%r)' % a + self._B__super.meth(a) vereq(B().meth(2), 'B(2)A(2)') class C(A): def meth(self, a): return 'C(%r)' % a + self._C__super.meth(a) C._C__super = super(C) vereq(C().meth(3), 'C(3)A(3)') class D(C, B): def meth(self, a): return 'D(%r)' % a + super(D, self).meth(a) vereq(D().meth(4), 'D(4)C(4)B(4)A(4)') class mysuper(super): def __init__(self, *args): return super(mysuper, self).__init__(*args) class E(D): def meth(self, a): return 'E(%r)' % a + mysuper(E, self).meth(a) vereq(E().meth(5), 'E(5)D(5)C(5)B(5)A(5)') class F(E): def meth(self, a): s = self._F__super return 'F(%r)[%s]' % (a, s.__class__.__name__) + s.meth(a) F._F__super = mysuper(F) vereq(F().meth(6), 'F(6)[mysuper]E(6)D(6)C(6)B(6)A(6)') try: super(D, 42) except TypeError: pass raise TestFailed, "shouldn't allow super(D, 42)" try: super(D, C()) except TypeError: pass raise TestFailed, "shouldn't allow super(D, C())" try: super(D).__get__(12) except TypeError: pass raise TestFailed, "shouldn't allow super(D).__get__(12)" try: super(D).__get__(C()) except TypeError: pass raise TestFailed, "shouldn't allow super(D).__get__(C())" class DDbase(object): def getx(self): return 42 x = property(getx) class DDsub(DDbase): def getx(self): return 'hello' x = property(getx) dd = DDsub() vereq(dd.x, 'hello') vereq(super(DDsub, dd).x, 42) class Base(object): aProp = property((lambda self: 'foo')) class Sub(Base): def test(klass): return super(Sub, klass).aProp test = classmethod(test) veris(Sub.test(), Base.aProp) def inherits(): if verbose: print 'Testing inheritance from basic types...' class hexint(int): def __repr__(self): return hex(self) def __add__(self, other): return hexint(int.__add__(self, other)) vereq(repr(hexint(7) + 9), '0x10') vereq(repr(hexint(1000) + 7), '0x3ef') a = hexint(12345) vereq(a, 12345) vereq(int(a), 12345) verify(int(a).__class__ is int) vereq(hash(a), hash(12345)) verify(+a.__class__ is int) verify((a >> 0).__class__ is int) verify((a << 0).__class__ is int) verify((hexint(0) << 12).__class__ is int) verify((hexint(0) >> 12).__class__ is int) class octlong(long): __slots__ = [] def __str__(self): s = oct(self) if s[-1] == 'L': s = s[:-1] return s def __add__(self, other): return self.__class__(super(octlong, self).__add__(other)) __radd__ = __add__ vereq(str(octlong(3) + 5), '010') vereq(str(5 + octlong(3000)), '05675') a = octlong(12345) vereq(a, 0x3039L) vereq(long(a), 0x3039L) vereq(hash(a), hash(0x3039L)) verify(long(a).__class__ is long) verify(+a.__class__ is long) verify(-a.__class__ is long) verify(-octlong(0).__class__ is long) verify((a >> 0).__class__ is long) verify((a << 0).__class__ is long) verify((a - 0).__class__ is long) verify((a * 1).__class__ is long) verify((a ** 1).__class__ is long) verify((a // 1).__class__ is long) verify((1 * a).__class__ is long) verify((a | 0).__class__ is long) verify((a ^ 0).__class__ is long) verify((a & -0x1L).__class__ is long) verify((octlong(0) << 12).__class__ is long) verify((octlong(0) >> 12).__class__ is long) verify(abs(octlong(0)).__class__ is long) class longclone(long): pass a = longclone(1) verify((a + 0).__class__ is long) verify((0 + a).__class__ is long) a = longclone(-1) vereq(a.__dict__, { }) vereq(long(a), -1) class precfloat(float): __slots__ = [ 'prec'] def __init__(self, value = 0.0, prec = 12): self.prec = int(prec) float.__init__(value) def __repr__(self): return '%.*g' % (self.prec, self) vereq(repr(precfloat(1.1000000000000001)), '1.1') a = precfloat(12345) vereq(a, 12345.0) vereq(float(a), 12345.0) verify(float(a).__class__ is float) vereq(hash(a), hash(12345.0)) verify(+a.__class__ is float) class madcomplex(complex): def __repr__(self): return '%.17gj%+.17g' % (self.imag, self.real) a = madcomplex(-3, 4) vereq(repr(a), '4j-3') base = complex(-3, 4) veris(base.__class__, complex) vereq(a, base) vereq(complex(a), base) veris(complex(a).__class__, complex) a = madcomplex(a) vereq(repr(a), '4j-3') vereq(a, base) vereq(complex(a), base) veris(complex(a).__class__, complex) vereq(hash(a), hash(base)) veris(+a.__class__, complex) veris((a + 0).__class__, complex) vereq(a + 0, base) veris((a - 0).__class__, complex) vereq(a - 0, base) veris((a * 1).__class__, complex) vereq(a * 1, base) veris((a / 1).__class__, complex) vereq(a / 1, base) class madtuple(tuple): _rev = None def rev(self): if self._rev is not None: return self._rev L = list(self) L.reverse() self._rev = self.__class__(L) return self._rev a = madtuple((1, 2, 3, 4, 5, 6, 7, 8, 9, 0)) vereq(a, (1, 2, 3, 4, 5, 6, 7, 8, 9, 0)) vereq(a.rev(), madtuple((0, 9, 8, 7, 6, 5, 4, 3, 2, 1))) vereq(a.rev().rev(), madtuple((1, 2, 3, 4, 5, 6, 7, 8, 9, 0))) for i in range(512): t = madtuple(range(i)) u = t.rev() v = u.rev() vereq(v, t) a = madtuple((1, 2, 3, 4, 5)) vereq(tuple(a), (1, 2, 3, 4, 5)) verify(tuple(a).__class__ is tuple) vereq(hash(a), hash((1, 2, 3, 4, 5))) verify(a[:].__class__ is tuple) verify((a * 1).__class__ is tuple) verify((a * 0).__class__ is tuple) verify((a + ()).__class__ is tuple) a = madtuple(()) vereq(tuple(a), ()) verify(tuple(a).__class__ is tuple) verify((a + a).__class__ is tuple) verify((a * 0).__class__ is tuple) verify((a * 1).__class__ is tuple) verify((a * 2).__class__ is tuple) verify(a[:].__class__ is tuple) class madstring(str): _rev = None def rev(self): if self._rev is not None: return self._rev L = list(self) L.reverse() self._rev = self.__class__(''.join(L)) return self._rev s = madstring('abcdefghijklmnopqrstuvwxyz') vereq(s, 'abcdefghijklmnopqrstuvwxyz') vereq(s.rev(), madstring('zyxwvutsrqponmlkjihgfedcba')) vereq(s.rev().rev(), madstring('abcdefghijklmnopqrstuvwxyz')) for i in range(256): s = madstring(''.join(map(chr, range(i)))) t = s.rev() u = t.rev() vereq(u, s) s = madstring('12345') vereq(str(s), '12345') verify(str(s).__class__ is str) base = '\x00' * 5 s = madstring(base) vereq(s, base) vereq(str(s), base) verify(str(s).__class__ is str) vereq(hash(s), hash(base)) vereq({ s: 1 }[base], 1) vereq({ base: 1 }[s], 1) verify((s + '').__class__ is str) vereq(s + '', base) verify(('' + s).__class__ is str) vereq('' + s, base) verify((s * 0).__class__ is str) vereq(s * 0, '') verify((s * 1).__class__ is str) vereq(s * 1, base) verify((s * 2).__class__ is str) vereq(s * 2, base + base) verify(s[:].__class__ is str) vereq(s[:], base) verify(s[0:0].__class__ is str) vereq(s[0:0], '') verify(s.strip().__class__ is str) vereq(s.strip(), base) verify(s.lstrip().__class__ is str) vereq(s.lstrip(), base) verify(s.rstrip().__class__ is str) vereq(s.rstrip(), base) identitytab = []([ chr(i) for i in range(256) ]) verify(s.translate(identitytab).__class__ is str) vereq(s.translate(identitytab), base) verify(s.translate(identitytab, 'x').__class__ is str) vereq(s.translate(identitytab, 'x'), base) vereq(s.translate(identitytab, '\x00'), '') verify(s.replace('x', 'x').__class__ is str) vereq(s.replace('x', 'x'), base) verify(s.ljust(len(s)).__class__ is str) vereq(s.ljust(len(s)), base) verify(s.rjust(len(s)).__class__ is str) vereq(s.rjust(len(s)), base) verify(s.center(len(s)).__class__ is str) vereq(s.center(len(s)), base) verify(s.lower().__class__ is str) vereq(s.lower(), base) class madunicode(unicode): _rev = None def rev(self): if self._rev is not None: return self._rev L = list(self) L.reverse() self._rev = self.__class__(u''.join(L)) return self._rev u = madunicode('ABCDEF') vereq(u, u'ABCDEF') vereq(u.rev(), madunicode(u'FEDCBA')) vereq(u.rev().rev(), madunicode(u'ABCDEF')) base = u'12345' u = madunicode(base) vereq(unicode(u), base) verify(unicode(u).__class__ is unicode) vereq(hash(u), hash(base)) vereq({ u: 1 }[base], 1) vereq({ base: 1 }[u], 1) verify(u.strip().__class__ is unicode) vereq(u.strip(), base) verify(u.lstrip().__class__ is unicode) vereq(u.lstrip(), base) verify(u.rstrip().__class__ is unicode) vereq(u.rstrip(), base) verify(u.replace(u'x', u'x').__class__ is unicode) vereq(u.replace(u'x', u'x'), base) verify(u.replace(u'xy', u'xy').__class__ is unicode) vereq(u.replace(u'xy', u'xy'), base) verify(u.center(len(u)).__class__ is unicode) vereq(u.center(len(u)), base) verify(u.ljust(len(u)).__class__ is unicode) vereq(u.ljust(len(u)), base) verify(u.rjust(len(u)).__class__ is unicode) vereq(u.rjust(len(u)), base) verify(u.lower().__class__ is unicode) vereq(u.lower(), base) verify(u.upper().__class__ is unicode) vereq(u.upper(), base) verify(u.capitalize().__class__ is unicode) vereq(u.capitalize(), base) verify(u.title().__class__ is unicode) vereq(u.title(), base) verify((u + u'').__class__ is unicode) vereq(u + u'', base) verify((u'' + u).__class__ is unicode) vereq(u'' + u, base) verify((u * 0).__class__ is unicode) vereq(u * 0, u'') verify((u * 1).__class__ is unicode) vereq(u * 1, base) verify((u * 2).__class__ is unicode) vereq(u * 2, base + base) verify(u[:].__class__ is unicode) vereq(u[:], base) verify(u[0:0].__class__ is unicode) vereq(u[0:0], u'') class sublist(list): pass a = sublist(range(5)) vereq(a, range(5)) a.append('hello') vereq(a, range(5) + [ 'hello']) a[5] = 5 vereq(a, range(6)) a.extend(range(6, 20)) vereq(a, range(20)) a[-5:] = [] vereq(a, range(15)) del a[10:15] vereq(len(a), 10) vereq(a, range(10)) vereq(list(a), range(10)) vereq(a[0], 0) vereq(a[9], 9) vereq(a[-10], 0) vereq(a[-1], 9) vereq(a[:5], range(5)) class CountedInput(file): '''Counts lines read by self.readline(). self.lineno is the 0-based ordinal of the last line read, up to a maximum of one greater than the number of lines in the file. self.ateof is true if and only if the final "" line has been read, at which point self.lineno stops incrementing, and further calls to readline() continue to return "". ''' lineno = 0 ateof = 0 def readline(self): if self.ateof: return '' s = file.readline(self) self.lineno += 1 if s == '': self.ateof = 1 return s f = file(name = TESTFN, mode = 'w') lines = [ 'a\n', 'b\n', 'c\n'] try: f.writelines(lines) f.close() f = CountedInput(TESTFN) for i, expected in zip(range(1, 5) + [ 4], lines + 2 * [ '']): got = f.readline() vereq(expected, got) vereq(f.lineno, i) vereq(f.ateof, i > len(lines)) f.close() finally: try: f.close() except: [] ''.join try: import os os.unlink(TESTFN) except: [] ''.join def keywords(): if verbose: print 'Testing keyword args to basic type constructors ...' vereq(int(x = 1), 1) vereq(float(x = 2), 2.0) vereq(long(x = 3), 0x3L) vereq(complex(imag = 42, real = 666), complex(666, 42)) vereq(str(object = 500), '500') vereq(unicode(string = 'abc', errors = 'strict'), u'abc') vereq(tuple(sequence = range(3)), (0, 1, 2)) vereq(list(sequence = (0, 1, 2)), range(3)) for constructor in (int, float, long, complex, str, unicode, tuple, list, file): try: constructor(bogus_keyword_arg = 1) except TypeError: continue raise TestFailed('expected TypeError from bogus keyword argument to %r' % constructor) def restricted(): return None import rexec if verbose: print 'Testing interaction with restricted execution ...' sandbox = rexec.RExec() code1 = "f = open(%r, 'w')" % TESTFN code2 = "f = file(%r, 'w')" % TESTFN code3 = "f = open(%r)\nt = type(f) # a sneaky way to get the file() constructor\nf.close()\nf = t(%r, 'w') # rexec can't catch this by itself\n" % (TESTFN, TESTFN) f = open(TESTFN, 'w') f.close() try: for code in (code1, code2, code3): try: sandbox.r_exec(code) except IOError: msg = None if str(msg).find('restricted') >= 0: outcome = 'OK' else: outcome = 'got an exception, but not an expected one' except: str(msg).find('restricted') >= 0 outcome = 'expected a restricted-execution exception' if outcome != 'OK': raise TestFailed('%s, in %r' % (outcome, code)) continue finally: try: import os os.unlink(TESTFN) except: pass def str_subclass_as_dict_key(): if verbose: print 'Testing a str subclass used as dict key ..' class cistr(str): '''Sublcass of str that computes __eq__ case-insensitively. Also computes a hash code of the string in canonical form. ''' def __init__(self, value): self.canonical = value.lower() self.hashcode = hash(self.canonical) def __eq__(self, other): if not isinstance(other, cistr): other = cistr(other) return self.canonical == other.canonical def __hash__(self): return self.hashcode vereq(cistr('ABC'), 'abc') vereq('aBc', cistr('ABC')) vereq(str(cistr('ABC')), 'ABC') d = { cistr('one'): 1, cistr('two'): 2, cistr('tHree'): 3 } vereq(d[cistr('one')], 1) vereq(d[cistr('tWo')], 2) vereq(d[cistr('THrEE')], 3) verify(cistr('ONe') in d) vereq(d.get(cistr('thrEE')), 3) def classic_comparisons(): if verbose: print 'Testing classic comparisons...' class classic: pass for base in (classic, int, object): if verbose: print ' (base = %s)' % base class C(base): def __init__(self, value): self.value = int(value) def __cmp__(self, other): if isinstance(other, C): return cmp(self.value, other.value) if isinstance(other, int) or isinstance(other, long): return cmp(self.value, other) return NotImplemented c1 = C(1) c2 = C(2) c3 = C(3) vereq(c1, 1) c = { 1: c1, 2: c2, 3: c3 } for x in (1, 2, 3): for y in (1, 2, 3): verify(cmp(c[x], c[y]) == cmp(x, y), 'x=%d, y=%d' % (x, y)) for op in ('<', '<=', '==', '!=', '>', '>='): verify(eval('c[x] %s c[y]' % op) == eval('x %s y' % op), 'x=%d, y=%d' % (x, y)) verify(cmp(c[x], y) == cmp(x, y), 'x=%d, y=%d' % (x, y)) verify(cmp(x, c[y]) == cmp(x, y), 'x=%d, y=%d' % (x, y)) def rich_comparisons(): if verbose: print 'Testing rich comparisons...' class Z(complex): pass z = Z(1) vereq(z, 1 + (0.0+0.0j)) vereq(1 + (0.0+0.0j), z) class ZZ(complex): def __eq__(self, other): try: return abs(self - other) <= 9.9999999999999995e-07 except: return NotImplemented zz = ZZ(1.0000003) vereq(zz, 1 + (0.0+0.0j)) vereq(1 + (0.0+0.0j), zz) class classic: pass for base in (classic, int, object, list): if verbose: print ' (base = %s)' % base class C(base): def __init__(self, value): self.value = int(value) def __cmp__(self, other): raise TestFailed, "shouldn't call __cmp__" def __eq__(self, other): if isinstance(other, C): return self.value == other.value if isinstance(other, int) or isinstance(other, long): return self.value == other return NotImplemented def __ne__(self, other): if isinstance(other, C): return self.value != other.value if isinstance(other, int) or isinstance(other, long): return self.value != other return NotImplemented def __lt__(self, other): if isinstance(other, C): return self.value < other.value if isinstance(other, int) or isinstance(other, long): return self.value < other return NotImplemented def __le__(self, other): if isinstance(other, C): return self.value <= other.value if isinstance(other, int) or isinstance(other, long): return self.value <= other return NotImplemented def __gt__(self, other): if isinstance(other, C): return self.value > other.value if isinstance(other, int) or isinstance(other, long): return self.value > other return NotImplemented def __ge__(self, other): if isinstance(other, C): return self.value >= other.value if isinstance(other, int) or isinstance(other, long): return self.value >= other return NotImplemented c1 = C(1) c2 = C(2) c3 = C(3) vereq(c1, 1) c = { 1: c1, 2: c2, 3: c3 } for x in (1, 2, 3): for y in (1, 2, 3): for op in ('<', '<=', '==', '!=', '>', '>='): verify(eval('c[x] %s c[y]' % op) == eval('x %s y' % op), 'x=%d, y=%d' % (x, y)) verify(eval('c[x] %s y' % op) == eval('x %s y' % op), 'x=%d, y=%d' % (x, y)) verify(eval('x %s c[y]' % op) == eval('x %s y' % op), 'x=%d, y=%d' % (x, y)) def coercions(): if verbose: print 'Testing coercions...' class I(int): pass coerce(I(0), 0) coerce(0, I(0)) class L(long): pass coerce(L(0), 0) coerce(L(0), 0x0L) coerce(0, L(0)) coerce(0x0L, L(0)) class F(float): pass coerce(F(0), 0) coerce(F(0), 0x0L) coerce(F(0), 0.0) coerce(0, F(0)) coerce(0x0L, F(0)) coerce(0.0, F(0)) class C(complex): pass coerce(C(0), 0) coerce(C(0), 0x0L) coerce(C(0), 0.0) coerce(C(0), (0.0+0.0j)) coerce(0, C(0)) coerce(0x0L, C(0)) coerce(0.0, C(0)) coerce((0.0+0.0j), C(0)) def descrdoc(): if verbose: print 'Testing descriptor doc strings...' def check(descr, what): vereq(descr.__doc__, what) check(file.closed, 'True if the file is closed') check(file.name, 'file name') def setclass(): if verbose: print 'Testing __class__ assignment...' class C(object): pass class D(object): pass class E(object): pass class F(D, E): pass for cls in (C, D, E, F): for cls2 in (C, D, E, F): x = cls() x.__class__ = cls2 verify(x.__class__ is cls2) x.__class__ = cls verify(x.__class__ is cls) def cant(x, C): try: x.__class__ = C except TypeError: pass raise TestFailed, "shouldn't allow %r.__class__ = %r" % (x, C) try: delattr(x, '__class__') except TypeError: pass raise TestFailed, "shouldn't allow del %r.__class__" % x cant(C(), list) cant(list(), C) cant(C(), 1) cant(C(), object) cant(object(), list) cant(list(), object) class Int(int): __slots__ = [] cant(2, Int) cant(Int(), int) cant(True, int) cant(2, bool) o = object() cant(o, type(1)) cant(o, type(None)) del o def setdict(): if verbose: print 'Testing __dict__ assignment...' class C(object): pass a = C() a.__dict__ = { 'b': 1 } vereq(a.b, 1) def cant(x, dict): try: x.__dict__ = dict except TypeError: pass raise TestFailed, "shouldn't allow %r.__dict__ = %r" % (x, dict) cant(a, None) cant(a, []) cant(a, 1) del a.__dict__ cant(C, { }) def pickles(): global C, C1, C2, C3, C4classic, C4 if verbose: print 'Testing pickling and copying new-style classes and objects...' import pickle import cPickle def sorteditems(d): L = d.items() L.sort() return L class C(object): def __init__(self, a, b): super(C, self).__init__() self.a = a self.b = b def __repr__(self): return 'C(%r, %r)' % (self.a, self.b) class C1(list): def __new__(cls, a, b): return super(C1, cls).__new__(cls) def __getnewargs__(self): return (self.a, self.b) def __init__(self, a, b): self.a = a self.b = b def __repr__(self): return 'C1(%r, %r)<%r>' % (self.a, self.b, list(self)) class C2(int): def __new__(cls, a, b, val = 0): return super(C2, cls).__new__(cls, val) def __getnewargs__(self): return (self.a, self.b, int(self)) def __init__(self, a, b, val = 0): self.a = a self.b = b def __repr__(self): return 'C2(%r, %r)<%r>' % (self.a, self.b, int(self)) class C3(object): def __init__(self, foo): self.foo = foo def __getstate__(self): return self.foo def __setstate__(self, foo): self.foo = foo class C4classic: pass class C4(C4classic, object): pass for p in (pickle, cPickle): for bin in (0, 1): if verbose: print p.__name__, [ 'text', 'binary'][bin] for cls in (C, C1, C2): s = p.dumps(cls, bin) cls2 = p.loads(s) verify(cls2 is cls) a = C1(1, 2) a.append(42) a.append(24) b = C2('hello', 'world', 42) s = p.dumps((a, b), bin) (x, y) = p.loads(s) vereq(x.__class__, a.__class__) vereq(sorteditems(x.__dict__), sorteditems(a.__dict__)) vereq(y.__class__, b.__class__) vereq(sorteditems(y.__dict__), sorteditems(b.__dict__)) vereq(repr(x), repr(a)) vereq(repr(y), repr(b)) if verbose: print 'a = x =', a print 'b = y =', b u = C3(42) s = p.dumps(u, bin) v = p.loads(s) veris(u.__class__, v.__class__) vereq(u.foo, v.foo) u = C4() u.foo = 42 s = p.dumps(u, bin) v = p.loads(s) veris(u.__class__, v.__class__) vereq(u.foo, v.foo) if verbose: print 'deepcopy' import copy for cls in (C, C1, C2): cls2 = copy.deepcopy(cls) verify(cls2 is cls) a = C1(1, 2) a.append(42) a.append(24) b = C2('hello', 'world', 42) (x, y) = copy.deepcopy((a, b)) vereq(x.__class__, a.__class__) vereq(sorteditems(x.__dict__), sorteditems(a.__dict__)) vereq(y.__class__, b.__class__) vereq(sorteditems(y.__dict__), sorteditems(b.__dict__)) vereq(repr(x), repr(a)) vereq(repr(y), repr(b)) if verbose: print 'a = x =', a print 'b = y =', b def pickleslots(): global B, C, D, C, D, E if verbose: print 'Testing pickling of classes with __slots__ ...' import pickle import cPickle class B(object): pass for base in [ object, B]: class C(base): __slots__ = [ 'a'] class D(C): pass try: pickle.dumps(C()) except TypeError: pass raise TestFailed, 'should fail: pickle C instance - %s' % base try: cPickle.dumps(C()) except TypeError: pass raise TestFailed, 'should fail: cPickle C instance - %s' % base try: pickle.dumps(C()) except TypeError: pass raise TestFailed, 'should fail: pickle D instance - %s' % base try: cPickle.dumps(D()) except TypeError: pass raise TestFailed, 'should fail: cPickle D instance - %s' % base class C(base): __slots__ = [ 'a'] def __getstate__(self): try: d = self.__dict__.copy() except AttributeError: d = { } for cls in self.__class__.__mro__: for sn in cls.__dict__.get('__slots__', ()): try: d[sn] = getattr(self, sn) continue except AttributeError: continue return d def __setstate__(self, d): for k, v in d.items(): setattr(self, k, v) class D(C): pass x = C() y = pickle.loads(pickle.dumps(x)) vereq(hasattr(y, 'a'), 0) y = cPickle.loads(cPickle.dumps(x)) vereq(hasattr(y, 'a'), 0) x.a = 42 y = pickle.loads(pickle.dumps(x)) vereq(y.a, 42) y = cPickle.loads(cPickle.dumps(x)) vereq(y.a, 42) x = D() x.a = 42 x.b = 100 y = pickle.loads(pickle.dumps(x)) vereq(y.a + y.b, 142) y = cPickle.loads(cPickle.dumps(x)) vereq(y.a + y.b, 142) class E(C): __slots__ = [ 'b'] x = E() x.a = 42 x.b = 'foo' y = pickle.loads(pickle.dumps(x)) vereq(y.a, x.a) vereq(y.b, x.b) y = cPickle.loads(cPickle.dumps(x)) vereq(y.a, x.a) vereq(y.b, x.b) def copies(): if verbose: print 'Testing copy.copy() and copy.deepcopy()...' import copy class C(object): pass a = C() a.foo = 12 b = copy.copy(a) vereq(b.__dict__, a.__dict__) a.bar = [ 1, 2, 3] c = copy.copy(a) vereq(c.bar, a.bar) verify(c.bar is a.bar) d = copy.deepcopy(a) vereq(d.__dict__, a.__dict__) a.bar.append(4) vereq(d.bar, [ 1, 2, 3]) def binopoverride(): if verbose: print 'Testing overrides of binary operations...' class I(int): def __repr__(self): return 'I(%r)' % int(self) def __add__(self, other): return I(int(self) + int(other)) __radd__ = __add__ def __pow__(self, other, mod = None): if mod is None: return I(pow(int(self), int(other))) else: return I(pow(int(self), int(other), int(mod))) def __rpow__(self, other, mod = None): if mod is None: return I(pow(int(other), int(self), mod)) else: return I(pow(int(other), int(self), int(mod))) vereq(repr(I(1) + I(2)), 'I(3)') vereq(repr(I(1) + 2), 'I(3)') vereq(repr(1 + I(2)), 'I(3)') vereq(repr(I(2) ** I(3)), 'I(8)') vereq(repr(2 ** I(3)), 'I(8)') vereq(repr(I(2) ** 3), 'I(8)') vereq(repr(pow(I(2), I(3), I(5))), 'I(3)') class S(str): def __eq__(self, other): return self.lower() == other.lower() def subclasspropagation(): if verbose: print 'Testing propagation of slot functions to subclasses...' class A(object): pass class B(A): pass class C(A): pass class D(B, C): pass d = D() vereq(hash(d), id(d)) A.__hash__ = lambda self: 42 vereq(hash(d), 42) C.__hash__ = lambda self: 314 vereq(hash(d), 314) B.__hash__ = lambda self: 144 vereq(hash(d), 144) D.__hash__ = lambda self: 100 vereq(hash(d), 100) del D.__hash__ vereq(hash(d), 144) del B.__hash__ vereq(hash(d), 314) del C.__hash__ vereq(hash(d), 42) del A.__hash__ vereq(hash(d), id(d)) d.foo = 42 d.bar = 42 vereq(d.foo, 42) vereq(d.bar, 42) def __getattribute__(self, name): if name == 'foo': return 24 return object.__getattribute__(self, name) A.__getattribute__ = __getattribute__ vereq(d.foo, 24) vereq(d.bar, 42) def __getattr__(self, name): if name in ('spam', 'foo', 'bar'): return 'hello' raise AttributeError, name B.__getattr__ = __getattr__ vereq(d.spam, 'hello') vereq(d.foo, 24) vereq(d.bar, 42) del A.__getattribute__ vereq(d.foo, 42) del d.foo vereq(d.foo, 'hello') vereq(d.bar, 42) del B.__getattr__ try: d.foo except AttributeError: pass raise TestFailed, 'd.foo should be undefined now' import gc class A(object): pass class B(A): pass del B gc.collect() A.__setitem__ = lambda *a: pass def buffer_inherit(): import binascii if verbose: print 'Testing that buffer interface is inherited ...' class MyStr(str): pass base = 'abc' m = MyStr(base) vereq(binascii.b2a_hex(m), binascii.b2a_hex(base)) class MyUni(unicode): pass base = u'abc' m = MyUni(base) vereq(binascii.b2a_hex(m), binascii.b2a_hex(base)) class MyInt(int): pass m = MyInt(42) try: binascii.b2a_hex(m) raise TestFailed('subclass of int should not have a buffer interface') except TypeError: pass def str_of_str_subclass(): import binascii import cStringIO if verbose: print 'Testing __str__ defined in subclass of str ...' class octetstring(str): def __str__(self): return binascii.b2a_hex(self) def __repr__(self): return self + ' repr' o = octetstring('A') vereq(type(o), octetstring) vereq(type(str(o)), str) vereq(type(repr(o)), str) vereq(ord(o), 65) vereq(str(o), '41') vereq(repr(o), 'A repr') vereq(o.__str__(), '41') vereq(o.__repr__(), 'A repr') capture = cStringIO.StringIO() print >>capture, o print >>capture, str(o) vereq(capture.getvalue(), '41\n41\n') capture.close() def kwdargs(): if verbose: print 'Testing keyword arguments to __init__, __call__...' def f(a): return a vereq(f.__call__(a = 42), 42) a = [] list.__init__(a, sequence = [ 0, 1, 2]) vereq(a, [ 0, 1, 2]) def delhook(): if verbose: print 'Testing __del__ hook...' log = [] class C(object): def __del__(self): log.append(1) c = C() vereq(log, []) del c vereq(log, [ 1]) class D(object): pass d = D() try: del d[0] except TypeError: pass raise TestFailed, "invalid del() didn't raise TypeError" def hashinherit(): if verbose: print 'Testing hash of mutable subclasses...' class mydict(dict): pass d = mydict() try: hash(d) except TypeError: pass raise TestFailed, 'hash() of dict subclass should fail' class mylist(list): pass d = mylist() try: hash(d) except TypeError: pass raise TestFailed, 'hash() of list subclass should fail' def strops(): try: 'a' + 5 except TypeError: pass raise TestFailed, "'' + 5 doesn't raise TypeError" try: ''.split('') except ValueError: pass raise TestFailed, "''.split('') doesn't raise ValueError" try: ''.join([ 0]) except TypeError: pass raise TestFailed, "''.join([0]) doesn't raise TypeError" try: ''.rindex('5') except ValueError: pass raise TestFailed, "''.rindex('5') doesn't raise ValueError" try: '%(n)s' % None except TypeError: pass raise TestFailed, "'%(n)s' % None doesn't raise TypeError" try: '%(n' % { } except ValueError: pass raise TestFailed, "'%(n' % {} '' doesn't raise ValueError" try: '%*s' % 'abc' except TypeError: pass raise TestFailed, "'%*s' % ('abc') doesn't raise TypeError" try: '%*.*s' % ('abc', 5) except TypeError: pass raise TestFailed, "'%*.*s' % ('abc', 5) doesn't raise TypeError" try: '%s' % (1, 2) except TypeError: pass raise TestFailed, "'%s' % (1, 2) doesn't raise TypeError" try: '%' % None except ValueError: pass raise TestFailed, "'%' % None doesn't raise ValueError" vereq('534253'.isdigit(), 1) vereq('534253x'.isdigit(), 0) vereq('%c' % 5, '\x05') vereq('%c' % '5', '5') def deepcopyrecursive(): if verbose: print 'Testing deepcopy of recursive objects...' class Node: pass a = Node() b = Node() a.b = b b.a = a z = deepcopy(a) def modules(): if verbose: print 'Testing uninitialized module objects...' M = ModuleType import types m = M.__new__(M) str(m) vereq(hasattr(m, '__name__'), 0) vereq(hasattr(m, '__file__'), 0) vereq(hasattr(m, 'foo'), 0) vereq(m.__dict__, None) m.foo = 1 vereq(m.__dict__, { 'foo': 1 }) def dictproxyiterkeys(): class C(object): def meth(self): pass if verbose: print 'Testing dict-proxy iterkeys...' keys = [ key for key in C.__dict__.iterkeys() ] keys.sort() vereq(keys, [ '__dict__', '__doc__', '__module__', '__weakref__', 'meth']) def dictproxyitervalues(): class C(object): def meth(self): pass if verbose: print 'Testing dict-proxy itervalues...' values = [ values for values in C.__dict__.itervalues() ] vereq(len(values), 5) def dictproxyiteritems(): class C(object): def meth(self): pass if verbose: print 'Testing dict-proxy iteritems...' keys = [ key for key, value in C.__dict__.iteritems() ] keys.sort() vereq(keys, [ '__dict__', '__doc__', '__module__', '__weakref__', 'meth']) def funnynew(): if verbose: print 'Testing __new__ returning something unexpected...' class C(object): def __new__(cls, arg): if isinstance(arg, str): return [ 1, 2, 3] elif isinstance(arg, int): return object.__new__(D) else: return object.__new__(cls) class D(C): def __init__(self, arg): self.foo = arg vereq(C('1'), [ 1, 2, 3]) vereq(D('1'), [ 1, 2, 3]) d = D(None) veris(d.foo, None) d = C(1) vereq(isinstance(d, D), True) vereq(d.foo, 1) d = D(1) vereq(isinstance(d, D), True) vereq(d.foo, 1) def imulbug(): if verbose: print 'Testing for __imul__ problems...' class C(object): def __imul__(self, other): return (self, other) x = C() y = x y *= 1.0 vereq(y, (x, 1.0)) y = x y *= 2 vereq(y, (x, 2)) y = x y *= 0x3L vereq(y, (x, 0x3L)) y = x y *= 0x1L << 100 vereq(y, (x, 0x1L << 100)) y = x y *= None vereq(y, (x, None)) y = x y *= 'foo' vereq(y, (x, 'foo')) def docdescriptor(): if verbose: print 'Testing __doc__ descriptor...' class DocDescr(object): def __get__(self, object, otype): if object: object = object.__class__.__name__ + ' instance' if otype: otype = otype.__name__ return 'object=%s; type=%s' % (object, otype) class OldClass: __doc__ = DocDescr() class NewClass(object): __doc__ = DocDescr() vereq(OldClass.__doc__, 'object=None; type=OldClass') vereq(OldClass().__doc__, 'object=OldClass instance; type=OldClass') vereq(NewClass.__doc__, 'object=None; type=NewClass') vereq(NewClass().__doc__, 'object=NewClass instance; type=NewClass') def string_exceptions(): if verbose: print 'Testing string exceptions ...' astring = 'An exception string.' try: raise astring except astring: pass raise TestFailed, 'builtin string not usable as exception' class MyStr(str): pass newstring = MyStr("oops -- shouldn't work") try: raise newstring except TypeError: pass except: raise TestFailed, 'string subclass allowed as exception' def copy_setstate(): if verbose: print 'Testing that copy.*copy() correctly uses __setstate__...' import copy class C(object): def __init__(self, foo = None): self.foo = foo self._C__foo = foo def setfoo(self, foo = None): self.foo = foo def getfoo(self): return self._C__foo def __getstate__(self): return [ self.foo] def __setstate__(self, lst): if not len(lst) == 1: raise AssertionError self._C__foo = self.foo = lst[0] a = C(42) a.setfoo(24) vereq(a.foo, 24) vereq(a.getfoo(), 42) b = copy.copy(a) vereq(b.foo, 24) vereq(b.getfoo(), 24) b = copy.deepcopy(a) vereq(b.foo, 24) vereq(b.getfoo(), 24) def slices(): if verbose: print 'Testing cases with slices and overridden __getitem__ ...' vereq('hello'[:4], 'hell') vereq('hello'[slice(4)], 'hell') vereq(str.__getitem__('hello', slice(4)), 'hell') class S(str): def __getitem__(self, x): return str.__getitem__(self, x) vereq(S('hello')[:4], 'hell') vereq(S('hello')[slice(4)], 'hell') vereq(S('hello').__getitem__(slice(4)), 'hell') vereq((1, 2, 3)[:2], (1, 2)) vereq((1, 2, 3)[slice(2)], (1, 2)) vereq(tuple.__getitem__((1, 2, 3), slice(2)), (1, 2)) class T(tuple): def __getitem__(self, x): return tuple.__getitem__(self, x) vereq(T((1, 2, 3))[:2], (1, 2)) vereq(T((1, 2, 3))[slice(2)], (1, 2)) vereq(T((1, 2, 3)).__getitem__(slice(2)), (1, 2)) vereq([ 1, 2, 3][:2], [ 1, 2]) vereq([ 1, 2, 3][slice(2)], [ 1, 2]) vereq(list.__getitem__([ 1, 2, 3], slice(2)), [ 1, 2]) class L(list): def __getitem__(self, x): return list.__getitem__(self, x) vereq(L([ 1, 2, 3])[:2], [ 1, 2]) vereq(L([ 1, 2, 3])[slice(2)], [ 1, 2]) vereq(L([ 1, 2, 3]).__getitem__(slice(2)), [ 1, 2]) a = L([ 1, 2, 3]) a[slice(1, 3)] = [ 3, 2] vereq(a, [ 1, 3, 2]) a[slice(0, 2, 1)] = [ 3, 1] vereq(a, [ 3, 1, 2]) a.__setitem__(slice(1, 3), [ 2, 1]) vereq(a, [ 3, 2, 1]) a.__setitem__(slice(0, 2, 1), [ 2, 3]) vereq(a, [ 2, 3, 1]) def subtype_resurrection(): if verbose: print 'Testing resurrection of new-style instance...' class C(object): container = [] def __del__(self): C.container.append(self) c = C() c.attr = 42 del c del C.container[-1] vereq(len(C.container), 1) vereq(C.container[-1].attr, 42) del C.__del__ def slottrash(): if verbose: print 'Testing slot trash...' class trash(object): __slots__ = [ 'x'] def __init__(self, x): self.x = x o = None for i in xrange(50000): o = trash(o) del o def slotmultipleinheritance(): class A(object): __slots__ = () class B(object): pass class C(A, B): __slots__ = () vereq(C.__basicsize__, B.__basicsize__) verify(hasattr(C, '__dict__')) verify(hasattr(C, '__weakref__')) C().x = 2 def testrmul(): if verbose: print 'Testing correct invocation of __rmul__...' class C(object): def __mul__(self, other): return 'mul' def __rmul__(self, other): return 'rmul' a = C() vereq(a * 2, 'mul') vereq(a * 2.2000000000000002, 'mul') vereq(2 * a, 'rmul') vereq(2.2000000000000002 * a, 'rmul') def testipow(): if verbose: print 'Testing correct invocation of __ipow__...' class C(object): def __ipow__(self, other): pass a = C() a **= 2 def do_this_first(): if verbose: print 'Testing SF bug 551412 ...' class UserLong(object): def __pow__(self, *args): pass try: pow(0x0L, UserLong(), 0x0L) except: pass if verbose: print 'Testing SF bug 570483...' type.mro(tuple) def test_mutable_bases(): if verbose: print 'Testing mutable bases...' class C(object): pass class C2(object): def __getattribute__(self, attr): if attr == 'a': return 2 else: return super(C2, self).__getattribute__(attr) def meth(self): return 1 class D(C): pass class E(D): pass d = D() e = E() D.__bases__ = (C,) D.__bases__ = (C2,) vereq(d.meth(), 1) vereq(e.meth(), 1) vereq(d.a, 2) vereq(e.a, 2) vereq(C2.__subclasses__(), [ D]) class L(list): pass try: L.__bases__ = (dict,) except TypeError: pass raise TestFailed, "shouldn't turn list subclass into dict subclass" try: list.__bases__ = (dict,) except TypeError: pass raise TestFailed, "shouldn't be able to assign to list.__bases__" try: del D.__bases__ except TypeError: pass raise TestFailed, "shouldn't be able to delete .__bases__" try: D.__bases__ = () except TypeError: msg = None if str(msg) == "a new-style class can't have only classic bases": raise TestFailed, 'wrong error message for .__bases__ = ()' except: str(msg) == "a new-style class can't have only classic bases" raise TestFailed, "shouldn't be able to set .__bases__ to ()" try: D.__bases__ = (D,) except TypeError: pass raise TestFailed, "shouldn't be able to create inheritance cycles" try: D.__bases__ = (C, C) except TypeError: pass raise TestFailed, "didn't detect repeated base classes" try: D.__bases__ = (E,) except TypeError: pass raise TestFailed, "shouldn't be able to create inheritance cycles" class Classic: def meth2(self): return 3 D.__bases__ = (C, Classic) vereq(d.meth2(), 3) vereq(e.meth2(), 3) try: d.a except AttributeError: pass raise TestFailed, 'attribute should have vanished' try: D.__bases__ = (Classic,) except TypeError: pass raise TestFailed, 'new-style class must have a new-style base' def test_mutable_bases_with_failing_mro(): if verbose: print 'Testing mutable bases with failing mro...' class WorkOnce(type): def __new__(self, name, bases, ns): self.flag = 0 return super(WorkOnce, self).__new__(WorkOnce, name, bases, ns) def mro(self): pass class WorkAlways(type): def mro(self): return type.mro(self) class C(object): pass class C2(object): pass class D(C): pass class E(D): pass class F(D): __metaclass__ = WorkOnce class G(D): __metaclass__ = WorkAlways E_mro_before = E.__mro__ D_mro_before = D.__mro__ try: D.__bases__ = (C2,) except RuntimeError: vereq(E.__mro__, E_mro_before) vereq(D.__mro__, D_mro_before) raise TestFailed, 'exception not propagated' def test_mutable_bases_catch_mro_conflict(): if verbose: print 'Testing mutable bases catch mro conflict...' class A(object): pass class B(object): pass class C(A, B): pass class D(A, B): pass class E(C, D): pass try: C.__bases__ = (B, A) except TypeError: pass raise TestFailed, "didn't catch MRO conflict" def mutable_names(): if verbose: print 'Testing mutable names...' class C(object): pass mod = C.__module__ C.__name__ = 'D' vereq((C.__module__, C.__name__), (mod, 'D')) C.__name__ = 'D.E' vereq((C.__module__, C.__name__), (mod, 'D.E')) def subclass_right_op(): if verbose: print 'Testing correct dispatch of subclass overloading __r<op>__...' class B(int): def __floordiv__(self, other): return 'B.__floordiv__' def __rfloordiv__(self, other): return 'B.__rfloordiv__' vereq(B(1) // 1, 'B.__floordiv__') vereq(1 // B(1), 'B.__rfloordiv__') class C(object): def __floordiv__(self, other): return 'C.__floordiv__' def __rfloordiv__(self, other): return 'C.__rfloordiv__' vereq(C() // 1, 'C.__floordiv__') vereq(1 // C(), 'C.__rfloordiv__') class D(C): def __floordiv__(self, other): return 'D.__floordiv__' def __rfloordiv__(self, other): return 'D.__rfloordiv__' vereq(D() // C(), 'D.__floordiv__') vereq(C() // D(), 'D.__rfloordiv__') class E(C): pass vereq(E.__rfloordiv__, C.__rfloordiv__) vereq(E() // 1, 'C.__floordiv__') vereq(1 // E(), 'C.__rfloordiv__') vereq(E() // C(), 'C.__floordiv__') vereq(C() // E(), 'C.__floordiv__') def dict_type_with_metaclass(): if verbose: print 'Testing type of __dict__ when __metaclass__ set...' class B(object): pass class M(type): pass class C: __metaclass__ = M veris(type(C.__dict__), type(B.__dict__)) def meth_class_get(): if verbose: print 'Testing __get__ method of METH_CLASS C methods...' arg = [ 1, 2, 3] res = { 1: None, 2: None, 3: None } vereq(dict.fromkeys(arg), res) vereq({ }.fromkeys(arg), res) descr = dict.__dict__['fromkeys'] vereq(descr.__get__(None, dict)(arg), res) vereq(descr.__get__({ })(arg), res) try: descr.__get__(None, None) except TypeError: pass raise TestFailed, "shouldn't have allowed descr.__get__(None, None)" try: descr.__get__(42) except TypeError: pass raise TestFailed, "shouldn't have allowed descr.__get__(42)" try: descr.__get__(None, 42) except TypeError: pass raise TestFailed, "shouldn't have allowed descr.__get__(None, 42)" try: descr.__get__(None, int) except TypeError: pass raise TestFailed, "shouldn't have allowed descr.__get__(None, int)" def isinst_isclass(): if verbose: print 'Testing proxy isinstance() and isclass()...' class Proxy(object): def __init__(self, obj): self._Proxy__obj = obj def __getattribute__(self, name): if name.startswith('_Proxy__'): return object.__getattribute__(self, name) else: return getattr(self._Proxy__obj, name) class C: pass a = C() pa = Proxy(a) verify(isinstance(a, C)) verify(isinstance(pa, C)) class D(C): pass a = D() pa = Proxy(a) verify(isinstance(a, C)) verify(isinstance(pa, C)) class C(object): pass a = C() pa = Proxy(a) verify(isinstance(a, C)) verify(isinstance(pa, C)) class D(C): pass a = D() pa = Proxy(a) verify(isinstance(a, C)) verify(isinstance(pa, C)) def proxysuper(): if verbose: print 'Testing super() for a proxy object...' class Proxy(object): def __init__(self, obj): self._Proxy__obj = obj def __getattribute__(self, name): if name.startswith('_Proxy__'): return object.__getattribute__(self, name) else: return getattr(self._Proxy__obj, name) class B(object): def f(self): return 'B.f' class C(B): def f(self): return super(C, self).f() + '->C.f' obj = C() p = Proxy(obj) vereq(C.__dict__['f'](p), 'B.f->C.f') def carloverre(): if verbose: print "Testing prohibition of Carlo Verre's hack..." try: object.__setattr__(str, 'foo', 42) except TypeError: pass raise TestFailed, 'Carlo Verre __setattr__ suceeded!' try: object.__delattr__(str, 'lower') except TypeError: pass raise TestFailed, 'Carlo Verre __delattr__ succeeded!' def weakref_segfault(): if verbose: print 'Testing weakref segfault...' import weakref class Provoker: def __init__(self, referrent): self.ref = weakref.ref(referrent) def __del__(self): x = self.ref() class Oops(object): pass o = Oops() o.whatever = Provoker(o) del o def filefault(): if verbose: print 'Testing sys.stdout is changed in getattr...' import sys class StdoutGuard: def __getattr__(self, attr): sys.stdout = sys.__stdout__ raise RuntimeError('Premature access to sys.stdout.%s' % attr) sys.stdout = StdoutGuard() try: print 'Oops!' except RuntimeError: pass def vicious_descriptor_nonsense(): if verbose: print 'Testing vicious_descriptor_nonsense...' class Evil(object): def __hash__(self): return hash('attr') def __eq__(self, other): del C.attr return 0 class Descr(object): def __get__(self, ob, type = None): return 1 class C(object): attr = Descr() c = C() c.__dict__[Evil()] = 0 vereq(c.attr, 1) import gc gc.collect() vereq(hasattr(c, 'attr'), False) import warnings def test_init(): class Foo(object): def __init__(self): return 10 oldfilters = warnings.filters[:] try: warnings.filterwarnings('error', category = RuntimeWarning) try: Foo() except RuntimeWarning: pass raise TestFailed, 'did not test __init__() for None return' finally: warnings.filters = oldfilters def test_main(): weakref_segfault() do_this_first() class_docstrings() lists() dicts() dict_constructor() test_dir() ints() longs() floats() complexes() spamlists() spamdicts() pydicts() pylists() metaclass() pymods() multi() mro_disagreement() diamond() ex5() monotonicity() consistency_with_epg() objects() slots() slotspecials() dynamics() errors() classmethods() classmethods_in_c() staticmethods() staticmethods_in_c() classic() compattr() newslot() altmro() overloading() methods() specials() weakrefs() properties() supers() inherits() keywords() restricted() str_subclass_as_dict_key() classic_comparisons() rich_comparisons() coercions() descrdoc() setclass() setdict() pickles() copies() binopoverride() subclasspropagation() buffer_inherit() str_of_str_subclass() kwdargs() delhook() hashinherit() strops() deepcopyrecursive() modules() dictproxyiterkeys() dictproxyitervalues() dictproxyiteritems() pickleslots() funnynew() imulbug() docdescriptor() string_exceptions() copy_setstate() slices() subtype_resurrection() slottrash() slotmultipleinheritance() testrmul() testipow() test_mutable_bases() test_mutable_bases_with_failing_mro() test_mutable_bases_catch_mro_conflict() mutable_names() subclass_right_op() dict_type_with_metaclass() meth_class_get() isinst_isclass() proxysuper() carloverre() filefault() vicious_descriptor_nonsense() test_init() if verbose: print 'All OK' if __name__ == '__main__': test_main()